iT邦幫忙

2018 iT 邦幫忙鐵人賽
DAY 6
0
AI & Machine Learning

深度學習環境安裝筆記系列 第 6

[Day 06] 深度學習環境安裝筆記 — 安裝 CUDA 與 cudnn

  • 分享至 

  • xImage
  •  

1.安裝 CUDA :

  1. 下載CUDA ,下面兩個都要下載

這邊建議裝 CUDA 8 就好了


2. 用以下指令安裝 CUDA

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt update && sudo apt install cuda -y
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb
  1. 編輯 ~/.bashrc ,加入 cuda 的環境變數,並用 source ~/.bashrc 更新環境變數
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/lib/nvidia-384                                                
export CUDA_HOME=/usr/local/cuda                                                          
export PATH=$PATH:/usr/local/cuda/bin
  1. 開啟新的終端機視窗,並複製 cuda 的 sample code ,在用以下指令測試cuda是否安裝成功
cd
cp -r /usr/local/cuda/samples ./cuda_samples
cd cuda_samples
make -j8

編譯完成後執行以下指令,如果最後出現Pass,代表安裝成功

./bin/x86_64/linux/release/deviceQuery

2.安裝 cudnn

  1. 下載cudnn,進入網站後點選 Download

    再來會要你登入帳號,如果沒有帳號就去註冊一個吧。

根據你的作業系統以及cuda版本來選擇要下載的安裝檔,總共有兩個要下載

  1. 用以下指令來安裝cudnn
tar xvfz cudnn-8.0-linux-x64-v7.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/* /usr/local/cuda/lib64
sudo dpkg -i libcudnn7-doc_7.0.5.15-1+cuda8.0_amd64.deb
  1. 複製 cudnn 的 sample code ,並用以下指令編譯並測試,如果最後出現Pass則cudnn安裝成功
cp -r /usr/src/cudnn_samples_v7 ./
cd cudnn_samples_v7/mnistCUDNN/
make -j8
./mnist
sudo apt install -f

到此 CUDA 跟 cudnn 就安裝完成了


上一篇
[Day 05] 深度學習環境安裝筆記 — 安裝 Nvidia 驅動程式
下一篇
[Day 07] 深度學習環境安裝筆記 — 安裝與編譯 OpenCV 01. OpenCV 2.4
系列文
深度學習環境安裝筆記13
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言